from selenium import webdriver
from options import options
import os


def insert_img(driver, name):
    dir_name = os.path.dirname(__file__)
    up_dir_name = os.path.dirname(dir_name)
    img_name = rf'{up_dir_name}\reports\screenshots\{name}.png'
    driver.get_screenshot_as_file(img_name)


if __name__ == '__main__':
    name = 'baidu'
    url = r'https://www.baidu.com/'
    driver = webdriver.Chrome(options=options)

    insert_img(driver, name)

    driver.quit()
